# -*- coding: utf-8 -*-
from pathlib import Path

# 基础配置
BASE_DIR = Path(__file__).parent.parent
DATA_DIR = BASE_DIR / "input_files"
STORAGE_DIR = BASE_DIR / "storage"

# Ollama配置
OLLAMA_CONFIG = {
    "base_url": "http://192.168.18.5:11434",
    "model": "qwen3:4b",
    "temperature": 0.3,
    "request_timeout": 120
}

# 嵌入模型配置
EMBEDDING_CONFIG = {
    "model_name": "BAAI/bge-base-zh-v1.5",
    "embed_batch_size": 32
}

# 文本分块配置
TEXT_SPLITTER_CONFIG = {
    "chunk_size": 512,
    "chunk_overlap": 64,
    "separator": "\n",
    "paragraph_separator": "\n\n"
}

# 查询引擎配置
QUERY_ENGINE_CONFIG = {
    "similarity_top_k": 3,
    "vector_store_query_mode": "hybrid",
    "alpha": 0.5,
    "verbose": True
}
